在正式進入 JAX 之前，老頭先整理了 JAX 學習資源，分享給大家。如果你有其他的好東西，請你也分享給老頭。

1. JAX reference documentation

https://jax.readthedocs.io/

這個是 JAX 官方的文件網站，其內容包括了：

GETTING STARTED
初學者的快速入門指引。「JAX 好好玩」大量的參考並引用這一部份的內容。老頭建議初次接觸 JAX 的讀者，能夠從頭到尾把它看懂。

REFERENCE DOCUMENTATION
針對 JAX 所使用的某些「技術」，或者 JAX 內某些「特定的功能」做詳細的說明。這些技術及功能，在 GETTING STARTED 裏大多都曾被提到，讀者如果不能夠完全了解，就可以到這裏來研讀細節。

老頭建議初學者先不用特意來讀這一部份，選擇性的來看就好了。例如在 GETTING STARTED 中會提到 Pytrees，如果你覺得不能完全掌握它，就可以讀一下 REFERENCE DOCUMENTATION 中對於 Pytrees 的詳細說明。

ADVANCED JAX TUTORIALS
進階 JAX 學習指南。學完了 GETTING STARTED ，如果你想對 JAX 有更深入的了解；或者你未來會在工作中使用 JAX，要拿 JAX 謀生，這一部份是必須讀的。在稍後「JAX 好好玩」也會將這一部份的內容介紹給大家。

NOTES
一些不屬於其他分類的雜項註記，在讀完 GETTING STARTED 之後，讀者可以快速的瀏覽一遍。

DEVELOPER DOCUMENTATION
JAX 開發者的相關文件，如果你立志成為 JAX 開發人員、想要直接由原始碼編譯程式庫、或是了解 JAX internal API，就必須讀一部份。「JAX 好好玩」並不會著墨這件文件。

API DOCUMENTATION
這個部份就不用多說了，老頭習慣把它開啓擺在桌面，研讀或寫碼的時候，隨時參考它。

2. GitHub: n2cholas/awesome-jax

(https://github.com/n2cholas/awesome-jax)

這是老頭目前為止所看到最完整的 JAX 資源列表，內容琳琅滿目、美不勝收，包含了：

Libraries
Models and Projects
Videos
Papers
Tutorials and Blog Posts
Community

要什麼東西，先到這裏去找就對了。

3. GitHub: Intro to ML using JAX

(https://github.com/deep-learning-indaba/indaba-pracs-2022/blob/main/practicals/Introduction_to_ML_using_JAX.ipynb )

這是一個玩家，在 GitHub 上分享的，他利用 JAX 來介紹機器學習，並且提供了樣本程式。重頭看一遍，讀者們必定獲益良多。

4. GitHub: google/jax

(https://github.com/google/jax)

JAX 是一個開源專案，它的原始碼存放在這裡。

5. Google JAX

(https://en.wikipedia.org/wiki/Google_JAX)

維基百科上關於 JAX 的說明，目前只有英文版本。